petrol pump python loop